package utility

import (
	"context"
	"github.com/dgrijalva/jwt-go"
	"github.com/gogf/gf/v2/os/gcfg"
	"sync"
	"time"
)

// 指定加密密钥
var jwtSecret = []byte("43286a86708820e38c333cdd4c496355")

// Claims Claim是一些实体（通常指的用户）的状态和额外的元数据
type Claims struct {
	UserId      uint   `json:"user_id"`
	UserAccount string `json:"user_account"`
	UserSalt    string `json:"user_salt"`

	jwt.StandardClaims
}

// Blacklist 黑名单存储
var blacklist = struct {
	sync.RWMutex
	tokens map[string]struct{}
}{
	tokens: make(map[string]struct{}),
}

// GenerateToken 根据用户的用户名和Salt产生token
func GenerateToken(ctx context.Context, userId uint, username, salt string) (string, error) {
	// 设置token有效时间
	nowTime := time.Now()
	expireTime := nowTime.Add(3 * time.Hour)

	claims := Claims{
		UserId:      userId,
		UserAccount: username,
		UserSalt:    salt,
		StandardClaims: jwt.StandardClaims{
			// 过期时间
			ExpiresAt: expireTime.Unix(),
			// 指定token发行人
			Issuer: "MoKoEnigma",
		},
	}

	tokenClaims := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
	// 该方法内部生成签名字符串，再用于获取完整、已签名的token
	tokenSecret, _ := gcfg.Instance().Get(ctx, "jwt.tokenSecret")
	jwtSecret = tokenSecret.Bytes()
	token, err := tokenClaims.SignedString(jwtSecret)
	return token, err
}

// ParseToken 根据传入的token值获取到Claims对象信息，（进而获取其中的用户名和密码）
func ParseToken(ctx context.Context, token string) (*Claims, error) {
	// 用于解析鉴权的声明，方法内部主要是具体的解码和校验的过程，最终返回*Token
	tokenClaims, err := jwt.ParseWithClaims(token, &Claims{}, func(token *jwt.Token) (interface{}, error) {
		tokenSecret, _ := gcfg.Instance().Get(ctx, "jwt.tokenSecret")
		jwtSecret = tokenSecret.Bytes()
		return jwtSecret, nil
	})

	if tokenClaims != nil {
		// 从tokenClaims中获取到Claims对象，并使用断言，将该对象转换为我们自己定义的Claims
		// 要传入指针，项目中结构体都是用指针传递，节省空间。
		if claims, ok := tokenClaims.Claims.(*Claims); ok && tokenClaims.Valid {
			// 检查黑名单
			blacklist.RLock()
			defer blacklist.RUnlock()
			if _, blacklisted := blacklist.tokens[token]; !blacklisted {
				return claims, nil
			}
		}
	}

	return nil, err
}

// ForbiddenToken 将JWT加入黑名单
func ForbiddenToken(token string) {
	blacklist.Lock()
	defer blacklist.Unlock()
	blacklist.tokens[token] = struct{}{}
}
